#!/usr/bin/env bash

# Train and test STAViS audiovisual models for all splits

base_path='./experiments'

split=1
python main.py --gpu_devices 0,1,2,3 --batch_size 128 --n_threads 8 \
    --dataset split${split} \
    --audiovisual True \
    --use_transformer \
    --nlevels 5 \
    --num_heads 8 \
    --attn_dropout 0.1 \
    --reduction 2 \
    --pool_layer tempAvg \
    --explain "transformer temporal SE" \
    --checkpoint 10 --n_epochs 60 \
    --root_path ${base_path} \
    --pretrain_path ./experiments/visual_SE_train_test/split1_results/save_60.pth \
    --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
  	--annotation_path_coutrot1_train ./data/fold_lists/Coutrot_db1_list_train_${split}_fps.txt \
  	--annotation_path_coutrot1_test ./data/fold_lists/Coutrot_db1_list_test_${split}_fps.txt \
  	--annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
  	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt \
  	--annotation_path_summe_train ./data/fold_lists/SumMe_list_train_${split}_fps.txt \
  	--annotation_path_summe_test ./data/fold_lists/SumMe_list_test_${split}_fps.txt \
  	--annotation_path_etmd_train ./data/fold_lists/ETMD_av_list_train_${split}_fps.txt \
  	--annotation_path_etmd_test ./data/fold_lists/ETMD_av_list_test_${split}_fps.txt \
  	--annotation_path_avad_train ./data/fold_lists/AVAD_list_train_${split}_fps.txt \
  	--annotation_path_avad_test ./data/fold_lists/AVAD_list_test_${split}_fps.txt

split=2
python main.py --gpu_devices 0,1 --batch_size 128 --n_threads 8 \
    --dataset split${split} \
    --audiovisual True \
    --use_transformer \
    --nlevels 5 \
    --num_heads 8 \
    --attn_dropout 0.1 \
    --reduction 2 \
    --pool_layer tempAvg \
    --explain "transformer temporal SE" \
    --checkpoint 10 --n_epochs 60 \
    --root_path ${base_path} \
    --pretrain_path ./experiments/visual_SE_train_test/split2_results/save_60.pth \
    --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
  	--annotation_path_coutrot1_train ./data/fold_lists/Coutrot_db1_list_train_${split}_fps.txt \
  	--annotation_path_coutrot1_test ./data/fold_lists/Coutrot_db1_list_test_${split}_fps.txt \
  	--annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
  	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt \
  	--annotation_path_summe_train ./data/fold_lists/SumMe_list_train_${split}_fps.txt \
  	--annotation_path_summe_test ./data/fold_lists/SumMe_list_test_${split}_fps.txt \
  	--annotation_path_etmd_train ./data/fold_lists/ETMD_av_list_train_${split}_fps.txt \
  	--annotation_path_etmd_test ./data/fold_lists/ETMD_av_list_test_${split}_fps.txt \
  	--annotation_path_avad_train ./data/fold_lists/AVAD_list_train_${split}_fps.txt \
  	--annotation_path_avad_test ./data/fold_lists/AVAD_list_test_${split}_fps.txt

split=3
python main.py --gpu_devices 0,1 --batch_size 128 --n_threads 8 \
    --dataset split${split} \
    --audiovisual True \
    --use_transformer \
    --nlevels 5 \
    --num_heads 8 \
    --attn_dropout 0.1 \
    --reduction 2 \
    --pool_layer tempAvg \
    --explain "transformer temporal SE" \
    --checkpoint 10 --n_epochs 60 \
    --root_path ${base_path} \
    --pretrain_path ./experiments/visual_SE_train_test/split3_results/save_60.pth \
    --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
  	--annotation_path_coutrot1_train ./data/fold_lists/Coutrot_db1_list_train_${split}_fps.txt \
  	--annotation_path_coutrot1_test ./data/fold_lists/Coutrot_db1_list_test_${split}_fps.txt \
  	--annotation_path_coutrot2_train ./data/fold_lists/Coutrot_db2_list_train_${split}_fps.txt \
  	--annotation_path_coutrot2_test ./data/fold_lists/Coutrot_db2_list_test_${split}_fps.txt \
  	--annotation_path_summe_train ./data/fold_lists/SumMe_list_train_${split}_fps.txt \
  	--annotation_path_summe_test ./data/fold_lists/SumMe_list_test_${split}_fps.txt \
  	--annotation_path_etmd_train ./data/fold_lists/ETMD_av_list_train_${split}_fps.txt \
  	--annotation_path_etmd_test ./data/fold_lists/ETMD_av_list_test_${split}_fps.txt \
  	--annotation_path_avad_train ./data/fold_lists/AVAD_list_train_${split}_fps.txt \
  	--annotation_path_avad_test ./data/fold_lists/AVAD_list_test_${split}_fps.txt
